D:\a\tools.proto\tools.proto\dynamic\src\proto\struct_ext.rs
Line | Count | Source |
1 | | // Copyright (c) 2025, BlockProject 3D |
2 | | // |
3 | | // All rights reserved. |
4 | | // |
5 | | // Redistribution and use in source and binary forms, with or without modification, |
6 | | // are permitted provided that the following conditions are met: |
7 | | // |
8 | | // * Redistributions of source code must retain the above copyright notice, |
9 | | // this list of conditions and the following disclaimer. |
10 | | // * Redistributions in binary form must reproduce the above copyright notice, |
11 | | // this list of conditions and the following disclaimer in the documentation |
12 | | // and/or other materials provided with the distribution. |
13 | | // * Neither the name of BlockProject 3D nor the names of its contributors |
14 | | // may be used to endorse or promote products derived from this software |
15 | | // without specific prior written permission. |
16 | | // |
17 | | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
18 | | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
19 | | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
20 | | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
21 | | // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
22 | | // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
23 | | // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
24 | | // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
25 | | // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
26 | | // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
27 | | // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
28 | | |
29 | | use bp3d_util::extension; |
30 | | use bp3d_protoc::compiler::structure::{Field, FieldType, Structure}; |
31 | | use crate::buffer::Builder; |
32 | | use crate::component::ComponentType; |
33 | | |
34 | | extension! { |
35 | | pub extension StructureExt: Structure { |
36 | | fn get_field(&self, path: &str) -> Option<&Field>; |
37 | | } |
38 | | } |
39 | | |
40 | 3 | pub(super) fn new_structure_internal<'a>(builder: Builder<'a>, value: &Structure) -> Builder<'a> { |
41 | 3 | let mut builder = builder.fixed(0, value.byte_size); |
42 | 13 | for field10 in &value.fields { |
43 | 10 | let mut field_builder = Builder::new(&field.name) |
44 | 10 | .fixed(field.loc.byte_offset, field.loc.byte_size); |
45 | 10 | if let Some(primitive) = crate::field::primitive::from_field(field) { Branch (45:16): [Folded - Ignored]
Branch (45:16): [True: 10, False: 0]
|
46 | 10 | field_builder = field_builder.primitive(primitive); |
47 | 10 | }0 |
48 | 10 | match &field.ty { |
49 | 0 | FieldType::Struct(v) => { |
50 | 0 | field_builder = field_builder.add_child(new_structure_internal(Builder::new(&field.name), v)); |
51 | 0 | } |
52 | 10 | _ => () |
53 | | } |
54 | 10 | builder = builder.add_child(field_builder); |
55 | | } |
56 | 3 | builder |
57 | 3 | } |
58 | | |
59 | | impl ComponentType for Structure { |
60 | 3 | fn name(&self) -> &str { |
61 | 3 | &self.name |
62 | 3 | } |
63 | | |
64 | 3 | fn build(&self, builder: Builder<'static>) -> Builder<'static> { |
65 | 3 | new_structure_internal(builder, self) |
66 | 3 | } |
67 | | } |
68 | | |
69 | | impl StructureExt for Structure { |
70 | 0 | fn get_field(&self, path: &str) -> Option<&Field> { |
71 | 0 | let mut segments = path.split('.'); |
72 | 0 | let mut structure = self; |
73 | 0 | while let Some(segment) = segments.next() { Branch (73:19): [Folded - Ignored]
Branch (73:19): [True: 0, False: 0]
|
74 | 0 | let field = structure.fields.iter().find(|v| v.name == segment)?; |
75 | 0 | match &field.ty { |
76 | 0 | FieldType::Struct(v) => structure = &*v, |
77 | | _ => { |
78 | 0 | match segments.next() { |
79 | 0 | None => return Some(field), |
80 | 0 | Some(_) => return None |
81 | | } |
82 | | } |
83 | | } |
84 | | } |
85 | 0 | None |
86 | 0 | } |
87 | | } |